import torchvision.models as models
import torch.nn as nn


class Cifar10ResNet(models.ResNet):
    def __init__(self):
        # 使用BasicBlock构建ResNet18，适用于CIFAR10数据集
        super().__init__(block=models.resnet.BasicBlock, 
            layers=[2, 2, 2, 2], num_classes=10)